#ifndef __GOFGAME_DUNGEONKEEPER_UI_UNIT_UPGRADE_EVENT_ARGS_H__
#define __GOFGAME_DUNGEONKEEPER_UI_UNIT_UPGRADE_EVENT_ARGS_H__

#include "gofgame/common/ui/ManualBubbleEventArgs.h"
#include "gofgame/dungeonkeeper/game/unit/UnitManager.h"

namespace gofgame {
	namespace dungeonkeeper {
		namespace ui {
			using namespace gofgame::common::ui;

			struct UnitUpgradeEventArgs : public ManualBubbleEventArgs {
			public:
				UnitUpgradeEventArgs(
					Component *sourceComponent,
					const UnitElementViewInfo& unitElementViewInfo) 
					: ManualBubbleEventArgs(sourceComponent),
					unitElementViewInfo(unitElementViewInfo) {
				}

				UnitElementViewInfo &getUnitElementViewInfo() {
					return this->unitElementViewInfo;
				}

			private:
				UnitElementViewInfo unitElementViewInfo;
			};
		}
	}
}

#endif //__GOFGAME_DUNGEONKEEPER_UI_UNIT_UPGRADE_EVENT_ARGS_H__